#include <vector>
#include <algorithm>
#include <numeric>
using namespace std;
class Solution
{
public:
    int numberOfArithmeticSlices(vector<int> &nums)
    {
        // 状态表示 dp[i] 以nums[i]为结尾的等差数列的个数
        // 转移方程 dp[i]=n[i]-n[i-1]==n[i-1]-n[i-2]?dp[i-1]+1:0;
        // 初始化 dp[0]=dp[1]=0
        // 返回值 accumulate(nums)
        vector<int> dp(nums.size(), 0);
        for (int i = 2; i < nums.size(); i++)
            if (nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2])
                dp[i] = dp[i - 1] + 1;
        return accumulate(dp.begin(), dp.end(), 0);
    }
    int numberOfArithmeticSlices(vector<int> &nums)
    {
        // 1 2 3 4 5
        //  3 1
        //  4 2+1
        //  5 3+2+1
        // dp[i] 以i为结尾等差数列的长度
        // if(n[i]-n[i-1]==s) dp[i]=dp[i-1]++;
        // else dp[i]=1; s=n[i];ans;
        vector<int> dp(nums.size(), 0);
        dp[0] = 0;
        int s = 0;
        bool f = false;
        int ans = 0;
        for (int i = 2; i < nums.size(); i++)
        {
            if (!f)
            {
                if (nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2])
                {
                    dp[i] = 3;
                    s = nums[i] - nums[i - 1];
                    f = true;
                }
            }
            else
            {
                if (nums[i] - nums[i - 1] == s)
                {
                    dp[i] = dp[i - 1] + 1;
                }
                else
                {
                    ans += ((1 + (dp[i - 1] - 2)) * (dp[i - 1] - 2)) / 2;
                    f = false;
                }
            }
        }
        if (f)
            ans += ((1 + (dp.back() - 2)) * (dp.back() - 2)) / 2;
        return ans;
    }
};